import torch

# 创建一个 2x3 的张量
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
print("创建的张量 x:")
print(x)
# 查看张量的形状
print("张量 x 的形状:", x.shape)
# 对张量进行加法操作
y = torch.tensor([[7, 8, 9], [10, 11, 12]])
z = x + y
print("张量相加的结果 z:")
print(z)
# 将张量移动到 GPU 上（如果可用）
if torch.cuda.is_available():
    x = x.cuda()
    y = y.cuda()
    z = x + y
    print("在 GPU 上计算的结果 z:")
    print(z)